pub trait Optimizer {
    fn step(&self);
    fn zero_grad(&self, set_to_none: bool);
}
